#! /usr/bin/python3

import pymysql

# 打开数据库连接
db = pymysql.connect("localhost", "root", "123456", "testdb")

# 使用cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()

# 使用execute()方法执行SQL，如果表存在则删除
cursor.execute('DROP TABLE IF EXISTS employee')

# 使用预处理语句创建表
sql = '''create table employee(
         id int primary key auto_increment,
         first_name char(20) not null,
         last_name char(20),
         age int(3),
         sex char(2),
         income float)'''
cursor.execute(sql)

# 关闭数据库连接
db.close()
